from modeltranslation.translator import translator, TranslationOptions

from django_generic_flatblocks.contrib.gblocks.models import *
from treenav.models import MenuItem


class MenuItemOptions(TranslationOptions):
    fields = ('label', )
translator.register(MenuItem, MenuItemOptions)


class TitleTranslationOptions(TranslationOptions):
    fields = ('title',)
translator.register(Title, TitleTranslationOptions)


class TextTranslationOptions(TranslationOptions):
    fields = ('text',)
translator.register(Text, TextTranslationOptions)


class TitleAndTextTranslationOptions(TranslationOptions):
    fields = ('title', 'text',)
translator.register(TitleAndText, TitleAndTextTranslationOptions)


class TitleTextAndImageTranslationOptions(TranslationOptions):
    fields = ('title', 'text',)
translator.register(TitleTextAndImage, TitleTextAndImageTranslationOptions)
